package net.zoneland.zrdp.framework.config.properties;

import javax.sql.DataSource;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Configuration;

import net.zoneland.jdbc.ISmartConfigDataSource;

/**
 * druid 配置属性
 *
 * @author zonevue
 */
@Configuration(proxyBeanMethods = false)
public class DruidProperties
{
    @Value("${spring.datasource.druid.initialSize}")
    private int initialSize;

    @Value("${spring.datasource.druid.minIdle}")
    private int minIdle;

    @Value("${spring.datasource.druid.maxActive}")
    private int maxActive;

    @Value("${spring.datasource.druid.maxWait}")
    private int maxWait;

    @Value("${spring.datasource.druid.connectTimeout}")
    private int connectTimeout;

    @Value("${spring.datasource.druid.socketTimeout}")
    private int socketTimeout;

    @Value("${spring.datasource.druid.timeBetweenEvictionRunsMillis}")
    private int timeBetweenEvictionRunsMillis;

    @Value("${spring.datasource.druid.minEvictableIdleTimeMillis}")
    private int minEvictableIdleTimeMillis;

    @Value("${spring.datasource.druid.maxEvictableIdleTimeMillis}")
    private int maxEvictableIdleTimeMillis;

    @Value("${spring.datasource.druid.validationQuery}")
    private String validationQuery;

    @Value("${spring.datasource.druid.testWhileIdle}")
    private boolean testWhileIdle;

    @Value("${spring.datasource.druid.testOnBorrow}")
    private boolean testOnBorrow;

    @Value("${spring.datasource.druid.testOnReturn}")
    private boolean testOnReturn;

    public ISmartConfigDataSource dataSource(final ISmartConfigDataSource datasource) {
        /* 配置初始化大小、最小、最大 */
        datasource.setInitialSize(initialSize);
        datasource.setMaxActive(maxActive);
        datasource.setMinIdle(minIdle);

        if (maxWait > 0) {
            /* 配置获取连接等待超时的时间 */
            datasource.setMaxWait(maxWait);
        }

        if (connectTimeout > 0) {
            /* 配置驱动连接超时时间，检测数据库建立连接的超时时间，单位是毫秒 */
            datasource.setConnectTimeout(connectTimeout);
        }
        if (socketTimeout > 0) {
            /* 配置网络超时时间，等待数据库操作完成的网络超时时间，单位是毫秒 */
            datasource.setSocketTimeout(socketTimeout);
        }
        if (timeBetweenEvictionRunsMillis > 0) {
            /* 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 */
            datasource.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);
        }
        if (minEvictableIdleTimeMillis > 0) {
            /* 配置一个连接在池中最小、最大生存的时间，单位是毫秒 */
            datasource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);
        }
        if (maxEvictableIdleTimeMillis > 0) {
            datasource.setMaxEvictableIdleTimeMillis(maxEvictableIdleTimeMillis);
        }
        /*
          用来检测连接是否有效的sql，要求是一个查询语句，常用select 'x'。如果validationQuery为null，testOnBorrow、testOnReturn、testWhileIdle都不会起作用。
         */
        datasource.setValidationQuery(validationQuery);
        if (testWhileIdle) {
            /* 建议配置为true，不影响性能，并且保证安全性。申请连接的时候检测，如果空闲时间大于timeBetweenEvictionRunsMillis，执行validationQuery检测连接是否有效。 */
            datasource.setTestWhileIdle(testWhileIdle);
        }
        if (testOnBorrow) {
            /* 申请连接时执行validationQuery检测连接是否有效，做了这个配置会降低性能。 */
            datasource.setTestOnBorrow(testOnBorrow);
        }
        if (testOnReturn) {
            /* 归还连接时执行validationQuery检测连接是否有效，做了这个配置会降低性能。 */
            datasource.setTestOnReturn(testOnReturn);
        }
        return datasource;
    }
}
